package main

import (
	"bufio"
	"fmt"
	"net"
)

//监听端口
//接收客户端请求建立链接
//创建goroutine处理链接。
func main() {
	// 作为服务端自己开辟一个地址和端口
	listen, err := net.Listen("tcp", "127.0.0.1:20000")
	if err != nil {
		fmt.Println("listen failed, err:", err)
		return
	}
	for {
		// 等待客户端连接
		conn, err := listen.Accept()
		if err != nil {
			fmt.Println("conn failed, err:", err)
			continue
		}
		fmt.Println("客户端", conn.RemoteAddr().String(), "连接成功")

		// 如果有客户端连接过来就开启一个goroutine去数据传输
		go process2(conn)
	}

}

//
func process(conn net.Conn) {
	// 关闭连接
	defer conn.Close()
	// 监听客户端连接
	for {
		reader := bufio.NewReader(conn)
		var buf [128]byte
		n, err := reader.Read(buf[:]) // 读取数据
		if err != nil {
			fmt.Println("read from client failed, err:", err)
			break
		}
		recvStr := string(buf[:n])
		fmt.Println("收到client端发来的数据：", recvStr)
		conn.Write([]byte(recvStr)) // 发送数据
	}
}

func process2(conn net.Conn) {
	defer conn.Close()
	for {
		// 首先作为服务端你要读取客户端的数据
		reader := bufio.NewReader(conn)
		// 创建一个容器把读到的数据放里面
		var buf [128]byte
		_, err := reader.Read(buf[:])
		if err != nil {
			fmt.Println("read error:", err)
			break
		}

		// 再把读到数据给发送出去
		conn.Write(buf[:])

	}
}
